package com.ai.module.websocket.vo;

import com.ai.module.mj.domain.Task;
import lombok.Data;

import java.util.List;

/**
 * @author: heJinQin
 * @create: 2024-03-30 13:12
 * @description:
 **/
@Data
public class WebSocketProduct {

    private AiImage aiImage;

    private AiChat aiChat;

    @Data
    public static class AiImage {

        private String taskId;

        private Task task;

        private String process;

        private int waiNumber;

    }

    @Data
    public static class AiChat {

        private List<String> answerList;

        private boolean end= false;

        private String message;

    }
}
